import tkinter as tk
from tkinter import filedialog, messagebox

def open_file_dialog():
    # 让用户选择多个图片文件
    file_paths = filedialog.askopenfilenames(
        title="选择图片文件",
        filetypes=[("图片文件", "*.jpg *.jpeg *.png *.bmp *.gif"), ("所有文件", "*.*")]
    )
    if file_paths:
        messagebox.showinfo("选择结果", f"已选择 {len(file_paths)} 个图片文件:\n" + "\n".join(file_paths))

def open_folder_dialog():
    # 让用户选择文件夹
    folder_path = filedialog.askdirectory(title="选择图片文件夹")
    if folder_path:
        # 这里可以添加遍历文件夹中图片文件的代码
        messagebox.showinfo("选择结果", f"已选择文件夹:\n{folder_path}\n(可以在此添加遍历文件夹中图片的功能)")

def button_click():
    # 创建新窗口作为图片导入界面
    import_window = tk.Toplevel(root)
    import_window.title("图片导入")
    import_window.geometry("400x250")
    
    # 添加说明标签
    label = tk.Label(import_window, text="请选择导入方式：")
    label.pack(pady=20)
    
    # 添加选择图片按钮
    file_button = tk.Button(
        import_window, 
        text="选择多个图片文件", 
        command=open_file_dialog,
        width=20,
        height=2
    )
    file_button.pack(pady=10)
    
    # 添加选择文件夹按钮
    folder_button = tk.Button(
        import_window, 
        text="选择图片文件夹", 
        command=open_folder_dialog,
        width=20,
        height=2
    )
    folder_button.pack(pady=10)
    
    # 添加提示标签
    hint_label = tk.Label(import_window, text="提示：选择多个图片文件时，可以按住Ctrl或Shift键多选")
    hint_label.pack(pady=10)

# 创建主窗口
root = tk.Tk()
root.title("图片翻译系统")
root.geometry("350x200")

# 添加标签
label = tk.Label(root, text="请选择导入的图片或文件夹（文件夹中只有图片）")
label.pack(pady=10)

# 添加按钮
button = tk.Button(root, text="导入", command=button_click)
button.pack(pady=20)

# 运行主循环
root.mainloop()